# -*- coding: utf-8 -*-
# @Time    : 2023/5/8 14:08
# @Author  : Pan
# @Software: PyCharm
# @Project : VisualFramework
# @FileName: __init__.py

from core.Diffusion import DiffusionEngine
from core.TargetDiffusion import TargetDiffusionEngine
from core.MAE import MAEEngine
from core.Clas import ClasEngine
from core.Image2Image import Image2ImageEngine
from core.SerrulaeImage import SerrulaeImageEngine
from core.SerrulaeNoise import SerrulaeNoiseEngine
from core.SerrulaeMutil import SerrulaeMutilEngine
from core.Pose import PoseEngine


def choose_core(config):
    return eval(config["type"]+"Engine")(config)
